/*
 * 描          述:  <描述>
 * 修  改   人:  PengQingyang
 * 修改时间:  2022年10月8日
 * <修改描述:>
 */
package com.tx.core.webdomain;

import java.net.UnknownHostException;
import java.util.LinkedList;
import java.util.List;
import java.util.Properties;

import com.tx.core.util.Arith;
import com.tx.core.util.IpUtils;
import com.tx.core.webdomain.server.Cpu;
import com.tx.core.webdomain.server.Jvm;
import com.tx.core.webdomain.server.Mem;
import com.tx.core.webdomain.server.Sys;
import com.tx.core.webdomain.server.SysFile;

import oshi.SystemInfo;
import oshi.hardware.CentralProcessor;
import oshi.hardware.CentralProcessor.TickType;
import oshi.hardware.GlobalMemory;
import oshi.hardware.HardwareAbstractionLayer;
import oshi.software.os.FileSystem;
import oshi.software.os.OSFileStore;
import oshi.software.os.OperatingSystem;
import oshi.util.Util;

/**
 * 服务器相关信息
 * <功能详细描述>
 * 
 * @author  PengQingyang
 * @version  [版本号, 2022年10月8日]
 * @see  [相关类/方法]
 * @since  [产品/模块版本]
 */
public class Server {
    
    private static final int OSHI_WAIT_SECOND = 1000;
    
    /** CPU相关信息 */
    private Cpu cpu = new Cpu();
    
    /** 內存相关信息 */
    private Mem mem = new Mem();
    
    /** JVM相关信息 */
    private Jvm jvm = new Jvm();
    
    /** 服务器相关信息 */
    private Sys sys = new Sys();
    
    /** 磁盘相关信息 */
    private List<SysFile> sysFiles = new LinkedList<SysFile>();
    
    /** <默认构造函数> */
    private Server() {
        super();
    }
    
    /**
     * @return 返回 cpu
     */
    public Cpu getCpu() {
        return cpu;
    }
    
    /**
     * @param 对cpu进行赋值
     */
    public void setCpu(Cpu cpu) {
        this.cpu = cpu;
    }
    
    /**
     * @return 返回 mem
     */
    public Mem getMem() {
        return mem;
    }
    
    /**
     * @param 对mem进行赋值
     */
    public void setMem(Mem mem) {
        this.mem = mem;
    }
    
    /**
     * @return 返回 jvm
     */
    public Jvm getJvm() {
        return jvm;
    }
    
    /**
     * @param 对jvm进行赋值
     */
    public void setJvm(Jvm jvm) {
        this.jvm = jvm;
    }
    
    /**
     * @return 返回 sys
     */
    public Sys getSys() {
        return sys;
    }
    
    /**
     * @param 对sys进行赋值
     */
    public void setSys(Sys sys) {
        this.sys = sys;
    }
    
    /**
     * @return 返回 sysFiles
     */
    public List<SysFile> getSysFiles() {
        return sysFiles;
    }
    
    /**
     * @param 对sysFiles进行赋值
     */
    public void setSysFiles(List<SysFile> sysFiles) {
        this.sysFiles = sysFiles;
    }
    
    /**
     * 对系统信息进行一次快照
     * <功能详细描述>
     * @throws Exception [参数说明]
     * 
     * @return void [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    public static Server snapshot() throws Exception {
        Server server = new Server();
        
        SystemInfo si = new SystemInfo();
        HardwareAbstractionLayer hal = si.getHardware();
        server.setCpuInfo(hal.getProcessor());
        server.setMemInfo(hal.getMemory());
        server.setSysInfo();
        server.setJvmInfo();
        server.setSysFiles(si.getOperatingSystem());
        
        return server;
    }
    
    /**
     * 设置CPU信息
     * <功能详细描述>
     * @param processor [参数说明]
     * 
     * @return void [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    private void setCpuInfo(CentralProcessor processor) {
        // CPU信息
        long[] prevTicks = processor.getSystemCpuLoadTicks();
        Util.sleep(OSHI_WAIT_SECOND);
        long[] ticks = processor.getSystemCpuLoadTicks();
        long nice = ticks[TickType.NICE.getIndex()]
                - prevTicks[TickType.NICE.getIndex()];
        long irq = ticks[TickType.IRQ.getIndex()]
                - prevTicks[TickType.IRQ.getIndex()];
        long softirq = ticks[TickType.SOFTIRQ.getIndex()]
                - prevTicks[TickType.SOFTIRQ.getIndex()];
        long steal = ticks[TickType.STEAL.getIndex()]
                - prevTicks[TickType.STEAL.getIndex()];
        long cSys = ticks[TickType.SYSTEM.getIndex()]
                - prevTicks[TickType.SYSTEM.getIndex()];
        long user = ticks[TickType.USER.getIndex()]
                - prevTicks[TickType.USER.getIndex()];
        long iowait = ticks[TickType.IOWAIT.getIndex()]
                - prevTicks[TickType.IOWAIT.getIndex()];
        long idle = ticks[TickType.IDLE.getIndex()]
                - prevTicks[TickType.IDLE.getIndex()];
        long totalCpu = user + nice + cSys + idle + iowait + irq + softirq
                + steal;
        cpu.setCpuNum(processor.getLogicalProcessorCount());
        cpu.setTotal(totalCpu);
        cpu.setSys(cSys);
        cpu.setUsed(user);
        cpu.setWait(iowait);
        cpu.setFree(idle);
    }
    
    /**
     * 设置内存信息
     * <功能详细描述>
     * @param memory [参数说明]
     * 
     * @return void [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    private void setMemInfo(GlobalMemory memory) {
        mem.setTotal(memory.getTotal());
        mem.setUsed(memory.getTotal() - memory.getAvailable());
        mem.setFree(memory.getAvailable());
    }
    
    /**
     * 设置服务器信息<br/>
     * <功能详细描述> [参数说明]
     * 
     * @return void [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    private void setSysInfo() {
        Properties props = System.getProperties();
        sys.setComputerName(IpUtils.getHostName());
        sys.setComputerIp(IpUtils.getHostIp());
        sys.setOsName(props.getProperty("os.name"));
        sys.setOsArch(props.getProperty("os.arch"));
        sys.setUserDir(props.getProperty("user.dir"));
    }
    
    /**
     * 设置Java虚拟机
     * <功能详细描述>
     * @throws UnknownHostException [参数说明]
     * 
     * @return void [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    private void setJvmInfo() throws UnknownHostException {
        Properties props = System.getProperties();
        jvm.setTotal(Runtime.getRuntime().totalMemory());
        jvm.setMax(Runtime.getRuntime().maxMemory());
        jvm.setFree(Runtime.getRuntime().freeMemory());
        jvm.setVersion(props.getProperty("java.version"));
        jvm.setHome(props.getProperty("java.home"));
    }
    
    /**
     * 设置磁盘信息
     * <功能详细描述>
     * @param os [参数说明]
     * 
     * @return void [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    private void setSysFiles(OperatingSystem os) {
        FileSystem fileSystem = os.getFileSystem();
        List<OSFileStore> fsArray = fileSystem.getFileStores();
        for (OSFileStore fs : fsArray) {
            long free = fs.getUsableSpace();
            long total = fs.getTotalSpace();
            long used = total - free;
            SysFile sysFile = new SysFile();
            sysFile.setDirName(fs.getMount());
            sysFile.setSysTypeName(fs.getType());
            sysFile.setTypeName(fs.getName());
            sysFile.setTotal(convertFileSize(total));
            sysFile.setFree(convertFileSize(free));
            sysFile.setUsed(convertFileSize(used));
            sysFile.setUsage(Arith.mul(Arith.div(used, total, 4), 100));
            sysFiles.add(sysFile);
        }
    }
    
    /**
     * 字节转换
     * <功能详细描述>
     * @param size
     * @return [参数说明]
     * 
     * @return String [返回类型说明]
     * @exception throws [异常类型] [异常说明]
     * @see [类、类#方法、类#成员]
     */
    private String convertFileSize(long size) {
        long kb = 1024;
        long mb = kb * 1024;
        long gb = mb * 1024;
        if (size >= gb) {
            return String.format("%.1f GB", (float) size / gb);
        } else if (size >= mb) {
            float f = (float) size / mb;
            return String.format(f > 100 ? "%.0f MB" : "%.1f MB", f);
        } else if (size >= kb) {
            float f = (float) size / kb;
            return String.format(f > 100 ? "%.0f KB" : "%.1f KB", f);
        } else {
            return String.format("%d B", size);
        }
    }
}
